from ..module import BaseModule
import nvtx

class NVTXHatSparse4D(BaseModule):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def _get_submodules(self):
        self.backbone = self.module.backbone
        self.neck = self.module.neck
        self.head = self.module.head
        self.submodules_conformed = True